package net.tools.jspupload;

import javax.servlet.ServletConfig;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.jsp.PageContext;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Vector;

/**
 * Date:2008-9-24<br>
 *
 * @author <a href="mailto:yanghongjian@htjs.net">YangHongJian</a>
 * @version 2.0
 * @since 1.4.2
 */
public class SmartUpload {
    protected byte m_binArray[];
    protected HttpServletRequest m_request;
    protected HttpServletResponse m_response;
    protected ServletContext m_application;
    private int m_totalBytes;
    private int m_currentIndex;
    private int m_startData;
    private int m_endData;
    private String m_boundary;
    private long m_totalMaxFileSize;
    private long m_maxFileSize;
    private Vector m_deniedFilesList;
    private Vector m_allowedFilesList;
    private boolean m_denyPhysicalPath;
    private String m_contentDisposition;
    public static final int SAVE_AUTO = 0;
    public static final int SAVE_VIRTUAL = 1;
    public static final int SAVE_PHYSICAL = 2;
    private Files m_files;
    private Request m_formRequest;

    public SmartUpload() {
        m_totalBytes = 0;
        m_currentIndex = 0;
        m_startData = 0;
        m_endData = 0;
        m_boundary = "";
        m_totalMaxFileSize = 0L;
        m_maxFileSize = 0L;
        m_deniedFilesList = new Vector();
        m_allowedFilesList = new Vector();
        m_denyPhysicalPath = false;
        m_contentDisposition = "";
        m_files = new Files();
        m_formRequest = new Request();
    }

    public final void init(ServletConfig servletconfig)
            throws ServletException {
        m_application = servletconfig.getServletContext();
    }

    public void service(HttpServletRequest httpservletrequest, HttpServletResponse httpservletresponse)
            throws ServletException, IOException {
        m_request = httpservletrequest;
        m_response = httpservletresponse;
    }

    public final void initialize(ServletConfig servletconfig, HttpServletRequest httpservletrequest, HttpServletResponse httpservletresponse)
            throws ServletException {
        m_application = servletconfig.getServletContext();
        m_request = httpservletrequest;
        m_response = httpservletresponse;
    }

    public final void initialize(PageContext pagecontext)
            throws ServletException {
        m_application = pagecontext.getServletContext();
        m_request = (HttpServletRequest) pagecontext.getRequest();
        m_response = (HttpServletResponse) pagecontext.getResponse();
    }

    public final void initialize(ServletContext servletcontext, HttpServletRequest httpservletrequest, HttpServletResponse httpservletresponse)
            throws ServletException {
        m_application = servletcontext;
        m_request = httpservletrequest;
        m_response = httpservletresponse;
    }

    public void upload()
            throws ServletException, IOException, SmartUploadException {
        int i = 0;
        long l = 0L;
        boolean flag1 = false;
        String s4 = "";
        String s5 = "";
        String s6 = "";
        String s7 = "";
        String s8 = "";
        String s9 = "";
        String s10 = "";
        m_totalBytes = m_request.getContentLength();
        m_binArray = new byte[m_totalBytes];
        int j;
        for (; i < m_totalBytes; i += j)
            try {
                m_request.getInputStream();
                j = m_request.getInputStream().read(m_binArray, i, m_totalBytes - i);
            }
            catch (Exception exception) {
                throw new SmartUploadException("Unable to upload.");
            }

        for (; !flag1 && m_currentIndex < m_totalBytes; m_currentIndex++)
            if (m_binArray[m_currentIndex] == 13)
                flag1 = true;
            else
                m_boundary = m_boundary + (char) m_binArray[m_currentIndex];

        if (m_currentIndex == 1)
            return;
        for (m_currentIndex++; m_currentIndex < m_totalBytes; m_currentIndex = m_currentIndex + 2) {
            String s1 = getDataHeader();
            m_currentIndex = m_currentIndex + 2;
            boolean flag3 = s1.indexOf("filename") > 0;
            String s3 = getDataFieldValue(s1, "name");
            if (flag3) {
                s6 = getDataFieldValue(s1, "filename");
                s4 = getFileName(s6);
                s5 = getFileExt(s4);
                s7 = getContentType(s1);
                s8 = getContentDisp(s1);
                s9 = getTypeMIME(s7);
                s10 = getSubTypeMIME(s7);
            }
            getDataSection();
            if (flag3 && s4.length() > 0) {
                if (m_deniedFilesList.contains(s5))
                    throw new SecurityException("The extension of the file is denied to be uploaded (1015).");
                if (!m_allowedFilesList.isEmpty() && !m_allowedFilesList.contains(s5))
                    throw new SecurityException("The extension of the file is not allowed to be uploaded (1010).");
                if (m_maxFileSize > 0L && (long) ((m_endData - m_startData) + 1) > m_maxFileSize)
                    throw new SecurityException("Size exceeded for this file : " + s4 + " (1105).");
                l += (m_endData - m_startData) + 1;
                if (m_totalMaxFileSize > 0L && l > m_totalMaxFileSize)
                    throw new SecurityException("Total File Size exceeded (1110).");
            }
            if (flag3) {
                File file = new File();
                file.setParent(this);
                file.setFieldName(s3);
                file.setFileName(s4);
                file.setFileExt(s5);
                file.setFilePathName(s6);
                file.setIsMissing(s6.length() == 0);
                file.setContentType(s7);
                file.setContentDisp(s8);
                file.setTypeMIME(s9);
                file.setSubTypeMIME(s10);
                if (s7.indexOf("application/x-macbinary") > 0)
                    m_startData = m_startData + 128;
                file.setSize((m_endData - m_startData) + 1);
                file.setStartData(m_startData);
                file.setEndData(m_endData);
                m_files.addFile(file);
            } else {
                String s11 = new String(m_binArray, m_startData, (m_endData - m_startData) + 1);
                m_formRequest.putParameter(s3, s11);
            }
            if ((char) m_binArray[m_currentIndex + 1] == '-')
                break;
        }

    }

    public int save(String s)
            throws ServletException, IOException, SmartUploadException {
        return save(s, 0);
    }

    public int save(String s, int i)
            throws ServletException, IOException, SmartUploadException {
        int j = 0;
        if (s == null)
            s = m_application.getRealPath("/");
        if (s.indexOf("/") != -1) {
            if (s.charAt(s.length() - 1) != '/')
                s = s + "/";
        } else if (s.charAt(s.length() - 1) != '\\')
            s = s + "\\";
        for (int k = 0; k < m_files.getCount(); k++)
            if (!m_files.getFile(k).isMissing()) {
                m_files.getFile(k).saveAs(s + m_files.getFile(k).getFileName(), i);
                j++;
            }

        return j;
    }

    public int getSize() {
        return m_totalBytes;
    }

    public byte getBinaryData(int i) {
        byte byte0;
        try {
            byte0 = m_binArray[i];
        }
        catch (Exception exception) {
            throw new ArrayIndexOutOfBoundsException("Index out of range (1005).");
        }
        return byte0;
    }

    public Files getFiles() {
        return m_files;
    }

    public Request getRequest() {
        return m_formRequest;
    }

    public void downloadFile(String s)
            throws ServletException, IOException, SmartUploadException {
        downloadFile(s, null, null);
    }

    public void downloadFile(String s, String s1)
            throws ServletException, IOException, SmartUploadException {
        downloadFile(s, s1, null);
    }

    public void downloadFile(String s, String s1, String s2)
            throws ServletException, IOException, SmartUploadException {
        downloadFile(s, s1, s2, 65000);
    }

    public void downloadFile(String s, String s1, String s2, int i)
            throws ServletException, IOException, SmartUploadException {
        if (s == null)
            throw new IllegalArgumentException("File '" + s + "' not found (1040).");
        if (s.equals(""))
            throw new IllegalArgumentException("File '" + s + "' not found (1040).");
        if (!isVirtual(s) && m_denyPhysicalPath)
            throw new SecurityException("Physical path is denied (1035).");
        if (isVirtual(s))
            s = m_application.getRealPath(s);
        java.io.File file = new java.io.File(s);
        FileInputStream fileinputstream = new FileInputStream(file);
        long l = file.length();
        int k = 0;
        byte abyte0[] = new byte[i];
        if (s1 == null)
            m_response.setContentType("application/x-msdownload");
        else if (s1.length() == 0)
            m_response.setContentType("application/x-msdownload");
        else
            m_response.setContentType(s1);
        m_response.setContentLength((int) l);
        m_contentDisposition = m_contentDisposition != null ? m_contentDisposition : "attachment;";
        if (s2 == null)
            m_response.setHeader("Content-Disposition", m_contentDisposition + " filename=" + getFileName(s));
        else if (s2.length() == 0)
            m_response.setHeader("Content-Disposition", m_contentDisposition);
        else
            m_response.setHeader("Content-Disposition", m_contentDisposition + " filename=" + s2);
        while ((long) k < l) {
            int j = fileinputstream.read(abyte0, 0, i);
            k += j;
            m_response.getOutputStream().write(abyte0, 0, j);
        }
        fileinputstream.close();
    }

    public void downloadField(ResultSet resultset, String s, String s1, String s2)
            throws ServletException, IOException, SQLException {
        if (resultset == null)
            throw new IllegalArgumentException("The RecordSet cannot be null (1045).");
        if (s == null)
            throw new IllegalArgumentException("The columnName cannot be null (1050).");
        if (s.length() == 0)
            throw new IllegalArgumentException("The columnName cannot be empty (1055).");
        byte abyte0[] = resultset.getBytes(s);
        if (s1 == null)
            m_response.setContentType("application/x-msdownload");
        else if (s1.length() == 0)
            m_response.setContentType("application/x-msdownload");
        else
            m_response.setContentType(s1);
        m_response.setContentLength(abyte0.length);
        if (s2 == null)
            m_response.setHeader("Content-Disposition", "attachment;");
        else if (s2.length() == 0)
            m_response.setHeader("Content-Disposition", "attachment;");
        else
            m_response.setHeader("Content-Disposition", "attachment; filename=" + s2);
        m_response.getOutputStream().write(abyte0, 0, abyte0.length);
    }

    public void fieldToFile(ResultSet resultset, String s, String s1)
            throws ServletException, IOException, SmartUploadException, SQLException {
        try {
            if (m_application.getRealPath(s1) != null)
                s1 = m_application.getRealPath(s1);
            InputStream inputstream = resultset.getBinaryStream(s);
            FileOutputStream fileoutputstream = new FileOutputStream(s1);
            int i;
            while ((i = inputstream.read()) != -1)
                fileoutputstream.write(i);
            fileoutputstream.close();
        }
        catch (Exception exception) {
            throw new SmartUploadException("Unable to save file from the DataBase (1020).");
        }
    }

    private String getDataFieldValue(String s, String s1) {
        String s2;
        String s3 = "";
        int i;
        s2 = s1 + "=" + '"';
        i = s.indexOf(s2);
        if (i > 0) {
            int j = i + s2.length();
            int k;
            k = j;
            s2 = "\"";
            int l = s.indexOf(s2, j);
            if (k > 0 && l > 0)
                s3 = s.substring(k, l);
        }
        return s3;
    }

    private String getFileExt(String s) {
        if (s == null)
            return null;
        String s1;
        int i = s.lastIndexOf(46) + 1;
        int j = s.length();
        s1 = s.substring(i, j);
        if (s.lastIndexOf(46) > 0)
            return s1;
        else
            return "";
    }

    private String getContentType(String s) {
        String s1;
        String s2 = "";
        s1 = "Content-Type:";
        int i = s.indexOf(s1) + s1.length();
        if (i != -1) {
            int j = s.length();
            s2 = s.substring(i, j);
        }
        return s2;
    }

    private String getTypeMIME(String s) {
        int i = s.indexOf("/");
        if (i != -1)
            return s.substring(1, i);
        else
            return s;
    }

    private String getSubTypeMIME(String s) {
        int i = s.indexOf("/") + 1;
        if (i != -1) {
            int j = s.length();
            return s.substring(i, j);
        } else {
            return s;
        }
    }

    private String getContentDisp(String s) {
        int i, j;
        i = s.indexOf(":") + 1;
        j = s.indexOf(";");
        return s.substring(i, j);
    }

    private void getDataSection() {
        int i = m_currentIndex;
        int j = 0;
        int k = m_boundary.length();
        m_startData = m_currentIndex;
        m_endData = 0;
        while (i < m_totalBytes)
            if (m_binArray[i] == (byte) m_boundary.charAt(j)) {
                if (j == k - 1) {
                    m_endData = ((i - k) + 1) - 3;
                    break;
                }
                i++;
                j++;
            } else {
                i++;
                j = 0;
            }
        m_currentIndex = m_endData + k + 3;
    }

    private String getDataHeader() {
        int i = m_currentIndex;
        int j = 0;
        for (boolean flag1 = false; !flag1;)
            if (m_binArray[m_currentIndex] == 13 && m_binArray[m_currentIndex + 2] == 13) {
                flag1 = true;
                j = m_currentIndex - 1;
                m_currentIndex = m_currentIndex + 2;
            } else {
                m_currentIndex++;
            }

        return new String(m_binArray, i, (j - i) + 1);
    }

    private String getFileName(String s) {
        int i;
        i = s.lastIndexOf(47);
        if (i != -1)
            return s.substring(i + 1, s.length());
        i = s.lastIndexOf(92);
        if (i != -1)
            return s.substring(i + 1, s.length());
        else
            return s;
    }

    public void setDeniedFilesList(String s)
            throws ServletException, IOException, SQLException {
        if (s != null) {
            String s2 = "";
            for (int i = 0; i < s.length(); i++)
                if (s.charAt(i) == ',') {
                    if (!m_deniedFilesList.contains(s2))
                        m_deniedFilesList.addElement(s2);
                    s2 = "";
                } else {
                    s2 = s2 + s.charAt(i);
                }

            if (!s2.equals(""))
                m_deniedFilesList.addElement(s2);
        } else {
            m_deniedFilesList = null;
        }
    }

    public void setAllowedFilesList(String s) {
        if (s != null) {
            String s2 = "";
            for (int i = 0; i < s.length(); i++)
                if (s.charAt(i) == ',') {
                    if (!m_allowedFilesList.contains(s2))
                        m_allowedFilesList.addElement(s2);
                    s2 = "";
                } else {
                    s2 = s2 + s.charAt(i);
                }

            if (!s2.equals(""))
                m_allowedFilesList.addElement(s2);
        } else {
            m_allowedFilesList = null;
        }
    }

    public void setDenyPhysicalPath(boolean flag) {
        m_denyPhysicalPath = flag;
    }

    public void setForcePhysicalPath() {
    }

    public void setContentDisposition(String s) {
        m_contentDisposition = s;
    }

    public void setTotalMaxFileSize(long l) {
        m_totalMaxFileSize = l;
    }

    public void setMaxFileSize(long l) {
        m_maxFileSize = l;
    }

    protected String getPhysicalPath(String s, int i)
            throws IOException {
        String s1 = "";
        String s2 = "";
        String s3;
        boolean flag = false;
        s3 = System.getProperty("file.separator");
        if (s == null)
            throw new IllegalArgumentException("There is no specified destination file (1140).");
        if (s.equals(""))
            throw new IllegalArgumentException("There is no specified destination file (1140).");
        if (s.lastIndexOf("\\") >= 0) {
            s1 = s.substring(0, s.lastIndexOf("\\"));
            s2 = s.substring(s.lastIndexOf("\\") + 1);
        }
        if (s.lastIndexOf("/") >= 0) {
            s1 = s.substring(0, s.lastIndexOf("/"));
            s2 = s.substring(s.lastIndexOf("/") + 1);
        }
        s1 = s1.length() != 0 ? s1 : "/";
        java.io.File file = new java.io.File(s1);
        if (file.exists())
            flag = true;
        if (i == 0) {
            if (isVirtual(s1)) {
                s1 = m_application.getRealPath(s1);
                if (s1.endsWith(s3))
                    s1 = s1 + s2;
                else
                    s1 = s1 + s3 + s2;
                return s1;
            }
            if (flag) {
                if (m_denyPhysicalPath)
                    throw new IllegalArgumentException("Physical path is denied (1125).");
                else
                    return s;
            } else {
                throw new IllegalArgumentException("This path does not exist (1135).");
            }
        }
        if (i == 1) {
            if (isVirtual(s1)) {
                s1 = m_application.getRealPath(s1);
                if (s1.endsWith(s3))
                    s1 = s1 + s2;
                else
                    s1 = s1 + s3 + s2;
                return s1;
            }
            if (flag)
                throw new IllegalArgumentException("The path is not a virtual path.");
            else
                throw new IllegalArgumentException("This path does not exist (1135).");
        }
        if (i == 2) {
            if (flag)
                if (m_denyPhysicalPath)
                    throw new IllegalArgumentException("Physical path is denied (1125).");
                else
                    return s;
            if (isVirtual(s1))
                throw new IllegalArgumentException("The path is not a physical path.");
            else
                throw new IllegalArgumentException("This path does not exist (1135).");
        } else {
            return null;
        }
    }

    public void uploadInFile(String s)
            throws IOException, SmartUploadException {
        int i;
        int j = 0;
        if (s == null)
            throw new IllegalArgumentException("There is no specified destination file (1025).");
        if (s.length() == 0)
            throw new IllegalArgumentException("There is no specified destination file (1025).");
        if (!isVirtual(s) && m_denyPhysicalPath)
            throw new SecurityException("Physical path is denied (1035).");
        i = m_request.getContentLength();
        m_binArray = new byte[i];
        int k;
        for (; j < i; j += k)
            try {
                k = m_request.getInputStream().read(m_binArray, j, i - j);
            }
            catch (Exception exception) {
                throw new SmartUploadException("Unable to upload.");
            }

        if (isVirtual(s))
            s = m_application.getRealPath(s);
        try {
            java.io.File file = new java.io.File(s);
            FileOutputStream fileoutputstream = new FileOutputStream(file);
            fileoutputstream.write(m_binArray);
            fileoutputstream.close();
        }
        catch (Exception exception1) {
            throw new SmartUploadException("The Form cannot be saved in the specified file (1030).");
        }
    }

    private boolean isVirtual(String s) {
        if (m_application.getRealPath(s) != null) {
            java.io.File file = new java.io.File(m_application.getRealPath(s));
            return file.exists();
        } else {
            return false;
        }
    }
}
